const { PrismaClient } = require('@prisma/client');

const prisma = new PrismaClient();

async function initCaseCategories() {
  try {
    console.log('开始初始化案例分类...');

    // 检查是否已有分类数据
    const existingCategories = await prisma.caseCategory.findMany();
    if (existingCategories.length > 0) {
      console.log('案例分类已存在，跳过初始化');
      return;
    }

    // 创建默认案例分类
    const categories = [
      {
        name: '电商零售',
        description: '电商和零售行业的AI应用案例',
        sort: 1,
      },
      {
        name: '内容创作',
        description: 'AI在内容创作领域的应用案例',
        sort: 2,
      },
      {
        name: '教育',
        description: 'AI在教育行业的应用案例',
        sort: 3,
      },
      {
        name: '金融科技',
        description: 'AI在金融科技领域的应用案例',
        sort: 4,
      },
      {
        name: '医疗健康',
        description: 'AI在医疗健康行业的应用案例',
        sort: 5,
      },
      {
        name: '制造业',
        description: 'AI在制造业的应用案例',
        sort: 6,
      },
      {
        name: '其他',
        description: '其他行业的AI应用案例',
        sort: 7,
      },
    ];

    for (const category of categories) {
      await prisma.caseCategory.create({
        data: category,
      });
      console.log(`创建分类: ${category.name}`);
    }

    console.log('案例分类初始化完成！');
  } catch (error) {
    console.error('初始化案例分类失败:', error);
  } finally {
    await prisma.$disconnect();
  }
}

initCaseCategories(); 